package com.biostime.qdingding.common.utils;

import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.util.Hashtable;

import javax.imageio.ImageIO;

import com.biostime.qdingding.common.config.Global;
import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.common.BitMatrix;

/**
 * 二维码生成工具
 * @author phil
 * @version 2015-11-24
 */
public class QrCodeUtils {
	
	/**
	 * 获取二维码
	 * @param content 二维码内容  
	 * @param width 二维码图片宽度 
	 * @param height 二维码图片高度  
	 * @param file 二维码的图片格式  
	 * @return
	 */
	public static String getCode (String content,int width,int height,String filename) throws Exception {
		
		if(StringUtils.isEmpty(filename) || filename.lastIndexOf(".")<0){
			throw new Exception("文件名为空或者格式不正确");
		}
		
		String path = null;
		
		Hashtable<EncodeHintType, String> hints = new Hashtable<EncodeHintType, String>();  
        hints.put(EncodeHintType.CHARACTER_SET, "utf-8");   // 内容所使用字符集编码  
          
        BitMatrix bitMatrix = new MultiFormatWriter().encode(content, BarcodeFormat.QR_CODE, width, height, hints);  
        
        path = "/qrcode/" + filename;
        
        //检查上传的物理路径是否存在，不存在则创建
        String fullUpLoadPath = Global.getUserfilesBaseDir() + Global.USERFILES_BASE_URL;
		FileUtils.createDirectory(FileUtils.path(fullUpLoadPath));
		
		FileUtils.createFile(fullUpLoadPath + path);
		
        String format = filename.substring(filename.lastIndexOf(".")+1);
        
        // 生成二维码  
        File outputFile = new File(fullUpLoadPath + path);
        MatrixToImageWriter.writeToFile(bitMatrix, format, outputFile);
        
        //路径修改（新的接口方式访问二维码图片）
        path = path.replaceFirst("/", "");
        path = path.replace("/", "-");
        path = path.replace(".", "-");
        
        return path;
	}
	
	
	/**
	 * 二维码的生成需要借助MatrixToImageWriter类，该类是由Google提供的，可以将该类直接拷贝到源码中使用
	 */
	public static class MatrixToImageWriter {
		private static final int BLACK = 0xFF000000;
		private static final int WHITE = 0xFFFFFFFF;
	
		private MatrixToImageWriter() {
		}
	
		public static BufferedImage toBufferedImage(BitMatrix matrix) {
			int width = matrix.getWidth();
			int height = matrix.getHeight();
			BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
			for (int x = 0; x < width; x++) {
				for (int y = 0; y < height; y++) {
					image.setRGB(x, y, matrix.get(x, y) ? BLACK : WHITE);
				}
			}
			return image;
		}
	
		public static void writeToFile(BitMatrix matrix, String format, File file) throws IOException {
			BufferedImage image = toBufferedImage(matrix);
			if (!ImageIO.write(image, format, file)) {
				throw new IOException("Could not write an image of format " + format + " to " + file);
			}
		}
	
		public static void writeToStream(BitMatrix matrix, String format, OutputStream stream) throws IOException {
			BufferedImage image = toBufferedImage(matrix);
			if (!ImageIO.write(image, format, stream)) {
				throw new IOException("Could not write an image of format " + format);
			}
		}
	}
}
